/* USER CODE BEGIN Header */
/**
  ******************************************************************************
  * @file    usart.c
  * @brief   This file provides code for the configuration
  *          of the USART instances.
  ******************************************************************************
  * @attention
  *
  * Copyright (c) 2025 STMicroelectronics.
  * All rights reserved.
  *
  * This software is licensed under terms that can be found in the LICENSE file
  * in the root directory of this software component.
  * If no LICENSE file comes with this software, it is provided AS-IS.
  *
  ******************************************************************************
  */
/* USER CODE END Header */
/* Includes ------------------------------------------------------------------*/
#include "usart.h"

/* USER CODE BEGIN 0 */

/* 定义缓冲区和状态变量 */
#define FLASH_PAGE_SIZE     1024    // STM32F103页大小为1K
#define UART_RX_BUFFER_SIZE FLASH_PAGE_SIZE

uint8_t uart_rx_buffer[UART_RX_BUFFER_SIZE];  // 串口接收缓冲区
uint32_t uart_rx_count = 0;                   // 接收计数器
uint32_t flash_write_address = APPLICATION_ADDRESS;  // 当前Flash写入地址
uint8_t is_flash_erased = 0;                  // Flash是否已擦除标志

/* USER CODE END 0 */

UART_HandleTypeDef huart1;

/* USART1 init function */

void MX_USART1_UART_Init(void)
{

  /* USER CODE BEGIN USART1_Init 0 */

  /* USER CODE END USART1_Init 0 */

  /* USER CODE BEGIN USART1_Init 1 */

  /* USER CODE END USART1_Init 1 */
  huart1.Instance = USART1;
  huart1.Init.BaudRate = 115200;
  huart1.Init.WordLength = UART_WORDLENGTH_8B;
  huart1.Init.StopBits = UART_STOPBITS_1;
  huart1.Init.Parity = UART_PARITY_NONE;
  huart1.Init.Mode = UART_MODE_TX_RX;
  huart1.Init.HwFlowCtl = UART_HWCONTROL_NONE;
  huart1.Init.OverSampling = UART_OVERSAMPLING_16;
  if (HAL_UART_Init(&huart1) != HAL_OK)
  {
    Error_Handler();
  }
  /* USER CODE BEGIN USART1_Init 2 */

  HAL_NVIC_SetPriority(USART1_IRQn, 0, 0);
  HAL_NVIC_EnableIRQ(USART1_IRQn);

  /* USER CODE END USART1_Init 2 */

}

void HAL_UART_MspInit(UART_HandleTypeDef* uartHandle)
{

  GPIO_InitTypeDef GPIO_InitStruct = {0};
  if(uartHandle->Instance==USART1)
  {
  /* USER CODE BEGIN USART1_MspInit 0 */

  /* USER CODE END USART1_MspInit 0 */
    /* USART1 clock enable */
    __HAL_RCC_USART1_CLK_ENABLE();

    __HAL_RCC_GPIOA_CLK_ENABLE();
    /**USART1 GPIO Configuration
    PA9     ------> USART1_TX
    PA10     ------> USART1_RX
    */
    GPIO_InitStruct.Pin = GPIO_PIN_9;
    GPIO_InitStruct.Mode = GPIO_MODE_AF_PP;
    GPIO_InitStruct.Speed = GPIO_SPEED_FREQ_HIGH;
    HAL_GPIO_Init(GPIOA, &GPIO_InitStruct);

    GPIO_InitStruct.Pin = GPIO_PIN_10;
    GPIO_InitStruct.Mode = GPIO_MODE_INPUT;
    GPIO_InitStruct.Pull = GPIO_NOPULL;
    HAL_GPIO_Init(GPIOA, &GPIO_InitStruct);

  /* USER CODE BEGIN USART1_MspInit 1 */

  /* USER CODE END USART1_MspInit 1 */
  }
}

void HAL_UART_MspDeInit(UART_HandleTypeDef* uartHandle)
{

  if(uartHandle->Instance==USART1)
  {
  /* USER CODE BEGIN USART1_MspDeInit 0 */

  /* USER CODE END USART1_MspDeInit 0 */
    /* Peripheral clock disable */
    __HAL_RCC_USART1_CLK_DISABLE();

    /**USART1 GPIO Configuration
    PA9     ------> USART1_TX
    PA10    ------> USART1_RX
    */
    HAL_GPIO_DeInit(GPIOA, GPIO_PIN_9|GPIO_PIN_10);

  /* USER CODE BEGIN USART1_MspDeInit 1 */

  /* USER CODE END USART1_MspDeInit 1 */
  }
}

/* USER CODE BEGIN 1 */

int fputc(int ch, FILE *f)
{
	HAL_UART_Transmit(&huart1, (uint8_t *)&ch, 1, 1000);
	return ch;
}


/* 串口接收中断回调函数 */
void HAL_UART_RxCpltCallback(UART_HandleTypeDef *huart)
{
    if (huart->Instance == USART1)
    {
        /* 将接收到的数据存入缓冲区 */
        uart_rx_buffer[uart_rx_count++] = *huart->pRxBuffPtr;
        
        /* 判断是否接收满一页(1K)数据 */
        if (uart_rx_count >= FLASH_PAGE_SIZE)
        {
            /* 擦除Flash (仅首次需要) */
            if (!is_flash_erased)
            {
                if (FLASH_EraseAppArea() == HAL_OK)
                {
                    is_flash_erased = 1;
                    printf("Flash erased successfully\r\n");
                }
                else
                {
                    printf("Flash erase failed\r\n");
                    uart_rx_count = 0;  // 重置接收计数器
                    HAL_UART_Receive_IT(huart, huart->pRxBuffPtr, 1);
                    return;
                }
            }
            
            /* 写入Flash */
            if (FLASH_WriteData(flash_write_address, uart_rx_buffer, FLASH_PAGE_SIZE) == HAL_OK)
            {
                printf("Written page to 0x%08lX\r\n", flash_write_address);
                flash_write_address += FLASH_PAGE_SIZE;  // 更新写入地址
                uart_rx_count = 0;  // 重置接收计数器
                
                /* 检查是否为最后一页（假设APP大小为3K，可根据实际修改） */
                if (flash_write_address >= (APPLICATION_ADDRESS + 3*FLASH_PAGE_SIZE))
                {
                    printf("Firmware update completed!\r\n");
                    printf("Rebooting to APP in 2 seconds...\r\n");
                    
                    /* 延时2秒确保上位机收到消息 */
                    HAL_Delay(2000);
                    
                    /* 跳转到APP */
                    iap_load_app(APPLICATION_ADDRESS);
                }
            }
            else
            {
                printf("Flash write error!\r\n");
                uart_rx_count = 0;  // 重置接收计数器
            }
        }
        
        /* 继续接收下一个字节 */
        HAL_UART_Receive_IT(huart, huart->pRxBuffPtr, 1);
    }
}

/* USER CODE END 1 */
